home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_StringIO.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-10-18  |  4.8 KB  |  139 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. import unittest
  5. import StringIO
  6. import cStringIO
  7. import types
  8. from test import test_support
  9.  
  10. class TestGenericStringIO(unittest.TestCase):
  11.     _line = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!'
  12.     constructor = str
  13.     
  14.     def setUp(self):
  15.         self._line = self.constructor(self._line)
  16.         self._lines = self.constructor((self._line + '\n') * 5)
  17.         self._fp = self.MODULE.StringIO(self._lines)
  18.  
  19.     
  20.     def test_reads(self):
  21.         eq = self.assertEqual
  22.         self.assertRaises(TypeError, self._fp.seek)
  23.         eq(self._fp.read(10), self._line[:10])
  24.         eq(self._fp.readline(), self._line[10:] + '\n')
  25.         eq(len(self._fp.readlines(60)), 2)
  26.  
  27.     
  28.     def test_writes(self):
  29.         f = self.MODULE.StringIO()
  30.         self.assertRaises(TypeError, f.seek)
  31.         f.write(self._line[:6])
  32.         f.seek(3)
  33.         f.write(self._line[20:26])
  34.         f.write(self._line[52])
  35.         self.assertEqual(f.getvalue(), 'abcuvwxyz!')
  36.  
  37.     
  38.     def test_writelines(self):
  39.         f = self.MODULE.StringIO()
  40.         f.writelines([
  41.             self._line[0],
  42.             self._line[1],
  43.             self._line[2]])
  44.         f.seek(0)
  45.         self.assertEqual(f.getvalue(), 'abc')
  46.  
  47.     
  48.     def test_writelines_error(self):
  49.         
  50.         def errorGen():
  51.             yield 'a'
  52.             raise KeyboardInterrupt()
  53.  
  54.         f = self.MODULE.StringIO()
  55.         self.assertRaises(KeyboardInterrupt, f.writelines, errorGen())
  56.  
  57.     
  58.     def test_truncate(self):
  59.         eq = self.assertEqual
  60.         f = self.MODULE.StringIO()
  61.         f.write(self._lines)
  62.         f.seek(10)
  63.         f.truncate()
  64.         eq(f.getvalue(), 'abcdefghij')
  65.         f.truncate(5)
  66.         eq(f.getvalue(), 'abcde')
  67.         f.write('xyz')
  68.         eq(f.getvalue(), 'abcdexyz')
  69.         f.close()
  70.         self.assertRaises(ValueError, f.write, 'frobnitz')
  71.  
  72.     
  73.     def test_closed_flag(self):
  74.         f = self.MODULE.StringIO()
  75.         self.assertEqual(f.closed, False)
  76.         f.close()
  77.         self.assertEqual(f.closed, True)
  78.         f = self.MODULE.StringIO('abc')
  79.         self.assertEqual(f.closed, False)
  80.         f.close()
  81.         self.assertEqual(f.closed, True)
  82.  
  83.     
  84.     def test_iterator(self):
  85.         eq = self.assertEqual
  86.         unless = self.failUnless
  87.         eq(iter(self._fp), self._fp)
  88.         unless(hasattr(self._fp, '__iter__'))
  89.         unless(hasattr(self._fp, 'next'))
  90.         i = 0
  91.         for line in self._fp:
  92.             eq(line, self._line + '\n')
  93.             i += 1
  94.         
  95.         eq(i, 5)
  96.  
  97.  
  98.  
  99. class TestStringIO(TestGenericStringIO):
  100.     MODULE = StringIO
  101.     
  102.     def test_unicode(self):
  103.         if not test_support.have_unicode:
  104.             return None
  105.         
  106.         f = self.MODULE.StringIO()
  107.         f.write(self._line[:6])
  108.         f.seek(3)
  109.         f.write(unicode(self._line[20:26]))
  110.         f.write(unicode(self._line[52]))
  111.         s = f.getvalue()
  112.         self.assertEqual(s, unicode('abcuvwxyz!'))
  113.         self.assertEqual(type(s), types.UnicodeType)
  114.  
  115.  
  116.  
  117. class TestcStringIO(TestGenericStringIO):
  118.     MODULE = cStringIO
  119.  
  120. import sys
  121. if sys.platform.startswith('java'):
  122.     buffer = str
  123.  
  124.  
  125. class TestBufferStringIO(TestStringIO):
  126.     constructor = buffer
  127.  
  128.  
  129. class TestBuffercStringIO(TestcStringIO):
  130.     constructor = buffer
  131.  
  132.  
  133. def test_main():
  134.     test_support.run_unittest(TestStringIO, TestcStringIO, TestBufferStringIO, TestBuffercStringIO)
  135.  
  136. if __name__ == '__main__':
  137.     test_main()
  138.  
  139.